from matplotlib.pylab import *
import numpy as np
from math import pi

if __name__=='__main__':

    fr0=np.loadtxt('real_soldm.dat');
    fi0=np.loadtxt('imag_soldm.dat');
    t=np.loadtxt('time.dat');

    fr1=np.loadtxt('rho_real.dat');
    fi1=np.loadtxt('rho_imag.dat');

    tau=np.loadtxt('tau.dat');

    clf();
    subplot(2,1,1);
    plot(t,fr0)
    plot(t,fr1,'--');
    subplot(2,1,2);
    plot(t,fi0);
    plot(t,fi1,'--');

    show()

    
    if (0):
        fr=np.loadtxt('real_sol.dat');
        fi=np.loadtxt('imag_sol.dat');
        f = np.sqrt(np.sum(fr**2+fi**2,axis=1));
    

        clf()
        
        plot(t,fr0[:,1],'--',t,fr[:,1]);
        #plot(t,fi,'--');
        #plot(t,f,'-.')
